// Copyright (C) 2021 THL A29 Limited, a Tencent company. All rights reserved.
//
// Licensed under the BSD 3-Clause License (the "License"); you may not use this
// file except in compliance with the License. You may obtain a copy of the
// License at
//
// https://opensource.org/licenses/BSD-3-Clause
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
// License for the specific language governing permissions and limitations under
// the License.

#include "yadcc/scheduler/running_task_bookkeeper.h"

#include <algorithm>
#include <mutex>
#include <string>
#include <vector>

namespace yadcc::scheduler {

void RunningTaskBookkeeper::SetServantRunningTasks(
    const std::string& servant_location, std::vector<RunningTask> tasks) {
  std::scoped_lock _(lock_);
  running_tasks_.erase(servant_location);
  running_tasks_.emplace(servant_location, std::move(tasks));
}

void RunningTaskBookkeeper::DropServant(const std::string& servant_location) {
  std::scoped_lock _(lock_);
  running_tasks_.erase(servant_location);
}

std::vector<RunningTask> RunningTaskBookkeeper::GetRunningTasks() const {
  std::vector<RunningTask> result;
  std::scoped_lock _(lock_);
  for (auto&& [k, v] : running_tasks_) {
    result.insert(result.begin(), v.begin(), v.end());
  }
  return result;
}

}  // namespace yadcc::scheduler
